package com.ophiux.utils.common;

import java.security.InvalidKeyException;  
import java.security.NoSuchAlgorithmException;  
import java.security.SecureRandom;
import java.security.Security;  
  
import javax.crypto.BadPaddingException;  
import javax.crypto.Cipher;  
import javax.crypto.IllegalBlockSizeException;  
import javax.crypto.KeyGenerator;  
import javax.crypto.NoSuchPaddingException;  
import javax.crypto.SecretKey;  
import javax.crypto.spec.SecretKeySpec;
  

/**
 * 数据加密
 * @author hhl
 *
 */
public class EncrypDES {  
      
    //KeyGenerator 提供对称密钥生成器的功能，支持各种算法  
    private KeyGenerator keygen;  
    //SecretKey 负责保存对称密钥  
    private SecretKey deskey;  
    //Cipher负责完成加密或解密工作  
    private Cipher c;  
    //该字节数组负责保存加密的结果  
    private byte[] cipherByte;  
      
    public EncrypDES() throws NoSuchAlgorithmException, NoSuchPaddingException{  
        Security.addProvider(new com.sun.crypto.provider.SunJCE());  
        //实例化支持DES算法的密钥生成器(算法名称命名需按规定，否则抛出异常)  
        keygen = KeyGenerator.getInstance("DES");  
        //生成密钥  
        deskey = keygen.generateKey();  
        //生成Cipher对象,指定其支持的DES算法  
        c = Cipher.getInstance("DES");  
    }  
      
    /** 
     * 对字符串加密 
     *  
     * @param str 
     * @return 
     * @throws InvalidKeyException 
     * @throws IllegalBlockSizeException 
     * @throws BadPaddingException 
     */  
    public byte[] Encrytor(String str) throws InvalidKeyException,  
            IllegalBlockSizeException, BadPaddingException {  
        // 根据密钥，对Cipher对象进行初始化，ENCRYPT_MODE表示加密模式  
        c.init(Cipher.ENCRYPT_MODE, deskey);  
        byte[] src = str.getBytes();  
        // 加密，结果保存进cipherByte  
        cipherByte = c.doFinal(src);  
        return cipherByte;  
    }  
  
    /** 
     * 对字符串解密 
     *  
     * @param buff 
     * @return 
     * @throws InvalidKeyException 
     * @throws IllegalBlockSizeException 
     * @throws BadPaddingException 
     */  
    public byte[] Decryptor(byte[] buff) throws InvalidKeyException,  
            IllegalBlockSizeException, BadPaddingException {  
        // 根据密钥，对Cipher对象进行初始化，DECRYPT_MODE表示加密模式  
        c.init(Cipher.DECRYPT_MODE, deskey);  
        cipherByte = c.doFinal(buff);  
        return cipherByte;  
    }  
  
    /** 
     * @param args 
     * @throws NoSuchPaddingException  
     * @throws NoSuchAlgorithmException  
     * @throws BadPaddingException  
     * @throws IllegalBlockSizeException  
     * @throws InvalidKeyException  
     */  
    public static void main(String[] args) throws Exception {  
//        EncrypDES de1 = new EncrypDES();  
//        String msg ="何=海-龙";  
//        byte[] encontent = de1.Encrytor(msg);  
//        byte[] decontent = de1.Decryptor(encontent);  
//        System.out.println("明文是:" + msg);  
//        System.out.println("加密后:" + new String(encontent).getBytes("utf-8"));  
//        System.out.println("解密后:" + new String(decontent));  
        
        EncrypDES de1 = new EncrypDES();  
        String msg ="1何=海-龙12312312322222222发是打发斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑" +
        		"夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什" +
        		"夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什" +
        		"顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬问题二姑夫施工方啊斯蒂芬阿什顿发斯蒂芬阿什顿发斯蒂芬" +
        		"问题二姑夫施工方啊方啊对萨芬撒地方阿什顿发发但是发送到发送到发是打发斯蒂芬爱的色放阿什顿发阿什顿发阿达发放";  
        byte[] encontent = de1.encrypt(msg,"111");  
        byte[] decontent = de1.decrypt(encontent,"111");  
        System.out.println("明文是:" + msg);  
        System.out.println("加密后:" + encontent.toString());  
        System.out.println("解密后:" + new String(decontent));  
    }  
    
    
    public static byte [] encrypt(String content, String password) {
        try {
          KeyGenerator kgen = KeyGenerator.getInstance("AES");
          kgen.init(128, new SecureRandom(password.getBytes()));
          SecretKey secretKey = kgen.generateKey();
          byte [] enCodeFormat = secretKey.getEncoded();
          SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");
          Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
          byte [] byteContent = content.getBytes("utf-8");
          cipher.init(Cipher.ENCRYPT_MODE, key);
          byte [] result = cipher.doFinal(byteContent);
          return result;
        }
        catch(Exception e) {
          e.printStackTrace();
        }
        return null;
      }

      public static byte [] decrypt(byte [] content, String password) {
        try {
          KeyGenerator kgen = KeyGenerator.getInstance("AES");
          kgen.init(128, new SecureRandom(password.getBytes()));
          SecretKey secretKey = kgen.generateKey();
          byte [] enCodeFormat = secretKey.getEncoded();
          SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");
          Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
          cipher.init(Cipher.DECRYPT_MODE, key);
          byte [] result = cipher.doFinal(content);
          return result;
        }
        catch(Exception e) {
        }
        return null;
      }
  
}  